package com.auto.netty.group;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.text.SimpleDateFormat;

public class GroupChatServerHandler extends SimpleChannelInboundHandler<String> {
    //定义一个Channel组用来管理所有的Channel
    //GlobalEventExecutor.INSTANCE 是全局的事件执行器，是一个单例
    private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
    SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

    /**
     * 该方法表示连接建立，一旦有连接，第一个被执行
     * 有连接，就将Channel交给ChannelGroup进行管理
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        //将该客户加入聊天的信息推送给其他在线的客户端
        /*
        writeAndFlush会将ChannelGroup中所有的Channel遍历，并发送消息
         */
        channelGroup.writeAndFlush("[客户端]" + channel.remoteAddress() + "加入聊天 " + sdf.format(new java.util.Date()) + "\n");
        //添加Channel
        channelGroup.add(channel);
    }

    /**
     * 当Channel处于活动状态，执行该方法，提示：XXX上线了
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress() + " 上线了~" + sdf.format(new java.util.Date()) + "\n");
    }

    /**
     * 当Channel处于不活动状态，执行该方法，提示：XXX下线了
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress() + " 下线了~" + sdf.format(new java.util.Date()) + "\n");
    }

    /**
     * 当连接断开时，执行该方法，提示：客户端XXX离开聊天
     * 在执行该方法是ChannelGroup会将执行了该方法的Channel进行移除
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        channelGroup.writeAndFlush("[客户端]" + ctx.channel().remoteAddress() + "离开聊天 " + sdf.format(new java.util.Date()) + "\n");
    }

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        //获取当前Channel
        Channel channel = ctx.channel();
        //遍历ChannelGroup
        channelGroup.forEach(ch -> {
            if (channel != ch) {
                //不是当前通道，转发消息
                ch.writeAndFlush("[客户端]" + channel.remoteAddress() + " 发送消息" + msg + sdf.format(new java.util.Date()) + "\n");
            } else {
                ch.writeAndFlush("[自己] 发送消息" + msg + sdf.format(new java.util.Date()) + "\n");
            }
        });
    }

    /**
     * 发送异常，关闭通道
     *
     * @param ctx
     * @param cause
     * @throws Exception
     */
    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        ctx.close();
    }
}
